import matplotlib.pyplot as plt
import numpy as np
import matplotlib as mpl
mpl.rcParams['font.sans-serif']=['SimHei']#让中文可以正常显示
plt.rcParams['axes.unicode_minus']=False #让负号能够正常显示
x=np.linspace(-5,5,50)
y1=2*x+1
y2=x**2
plt.figure()
plt.subplot(2,1,1) #表示整个图像分割成2行1列，当前位置为1
plt.plot(x,y1,'c--',label='y1=2x+1')
plt.legend(loc='best')
plt.subplot(2,1,2) #表示整个图像分割成2行1列，当前位置为2
plt.plot(x,y2,'k-.',label='y2=x^2')
plt.legend(loc='best')
plt.show()

# 显示多张图
x=np.linspace(-5,5,50)
y=[2*x+1,x**2,3*x,x**2+1]
title_text=['y=2*x+1','y=x**2','y=3*x','y=x**2+1']
plt.figure()
for i in range(4):
    plt.subplot(2,2,i+1)
    plt.plot(x,y[i])
    plt.title(title_text[i])
plt.tight_layout()  #自动调整子图参数，使子图显示完整。
plt.show()
